package org.light.verb;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

import org.light.core.Verb;
import org.light.core.Writeable;
import org.light.domain.Domain;
import org.light.domain.Dropdown;
import org.light.domain.Field;
import org.light.domain.JavascriptBlock;
import org.light.domain.JavascriptMethod;
import org.light.domain.Method;
import org.light.domain.Signature;
import org.light.domain.Statement;
import org.light.domain.StatementList;
import org.light.domain.Type;
import org.light.easyui.EasyUIPositions;
import org.light.exception.ValidateException;
import org.light.oracle.generator.MybatisOracleSqlReflector;
import org.light.utils.DomainUtil;
import org.light.utils.MybatisSqlReflector;
import org.light.utils.PgsqlReflector;
import org.light.utils.StringUtil;
import org.light.utils.WriteableUtil;

public class Add extends Verb implements EasyUIPositions {
	protected Set<Field> deniedFields = new TreeSet<>();

	@Override
	public Method generateDaoImplMethod() throws Exception {
		if (this.denied)
			return null;
		else {
			Method method = new Method();
			method.setStandardName("Add" + StringUtil.capFirst(this.domain.getStandardName()));
			method.setNoContainer(false);
			method.addSignature(new Signature(1, "&self",""));
			method.addSignature(
					new Signature(2, StringUtil.getSnakeName(this.domain.getStandardName()), new Type(this.domain.getCapFirstDomainNameWithSuffix())));
			method.setReturnType(new Type("Result<u64, sqlx::Error>"));
			
			List<Writeable> sList = new ArrayList<Writeable>();	
			Set<String> deniedFieldNames = this.getDeniedFieldNames(this.deniedFields);
			
			sList.add(new Statement(500L,2,"let result = sqlx::query("));
			if (StringUtil.isBlank(this.getDbType())||this.getDbType().equalsIgnoreCase("MariaDB")||this.getDbType().equalsIgnoreCase("MySQL")) {
				sList.add(new Statement(1000L,2,"r#\""+MybatisSqlReflector.generateInsertSqlWithDeniedFields(domain,this.deniedFields)+"\"#"));
			}else if (this.getDbType().equalsIgnoreCase("PostgreSQL")||this.getDbType().equalsIgnoreCase("pgsql")) {
				sList.add(new Statement(1000L,2,"r#\""+ PgsqlReflector.generateInsertSqlWithDeniedFields(domain,this.deniedFields)+"\"#"));
			}else if (this.getDbType().equalsIgnoreCase("Oracle")) {
				sList.add(new Statement(1000L,2,"r#\""+ MybatisOracleSqlReflector.generateInsertSqlWithDeniedFields(domain,this.deniedFields)+"\"#"));
			}			
			sList.add(new Statement(2000L,2,")"));
			long serial = 3000L;
			for (Field f:domain.getFieldsWithoutId()) {
				if (!deniedFieldNames.contains(f.getFieldName())) {
					sList.add(new Statement(serial,2,".bind("+this.domain.getSnakeDomainName()+"."+f.getSnakeFieldName()+")"));
					serial += 1000L;
				}
			}
			sList.add(new Statement(serial+2000L,2,".execute(&*self.pool)"));
			sList.add(new Statement(serial+3000L,2,".await;"));
			sList.add(new Statement(serial+10000L,2,"self.pool.close();"));
			sList.add(new Statement(serial+11000L,2,"return result;"));

			method.setMethodStatementList(WriteableUtil.merge(sList));
			return method;
		}
	}

	@Override
	public Method generateDaoMethodDefinition() throws Exception {
		if (this.denied)
			return null;
		else {
			Method method = new Method();
			method.setStandardName("add" + StringUtil.capFirst(this.domain.getStandardName()));
			method.addSignature(
					new Signature(1, StringUtil.lowerFirst(this.domain.getStandardName()), this.domain.getType()));
			return method;
		}
	}

	@Override
	public Method generateServiceMethodDefinition() throws Exception {
		if (this.denied)
			return null;
		else {
			Method method = new Method();
			method.setStandardName("Add" + StringUtil.capFirst(this.domain.getStandardName()));
			method.addSignature(
					new Signature(1, StringUtil.lowerFirst(this.domain.getStandardName()), this.domain.getType()));
			return method;
		}
	}

	@Override
	public Method generateServiceImplMethod() throws Exception {
		if (this.denied)
			return null;
		else {
			Method method = new Method();
			method.setStandardName("Add" + StringUtil.capFirst(this.domain.getStandardName()));
			method.addSignature(
					new Signature(1, this.domain.getSnakeDomainName(), new Type(this.domain.getCapFirstDomainNameWithSuffix())));
			method.setReturnType(new Type("Result<u64, sqlx::Error>"));
			List<Writeable> sList = new ArrayList<Writeable>();
		
			sList.add(new Statement(1000L,1,"let app_state = init_db();"));
			sList.add(new Statement(2000L,1,"app_state.await.context."+StringUtil.getSnakeName(this.domain.getPlural())+"."+StringUtil.getSnakeName(this.getVerbName())+"("+this.domain.getSnakeDomainName()+").await"));

			method.setMethodStatementList(WriteableUtil.merge(sList));
			return method;
		}
	}
	public Add() {
		super();
		this.setLabel("新增");
	}

	public Add(Domain domain) throws ValidateException {
		super();
		this.domain = domain;
		this.denied = domain.isVerbDenied("Add");
		this.setVerbName("Add" + StringUtil.capFirst(this.domain.getStandardName()));
		this.setLabel("新增");
		if (domain.getLanguage().equalsIgnoreCase("english"))
			this.setLabel("Add");
	}

	@Override
	public Method generateControllerMethod() throws Exception {
		if (this.denied)
			return null;
		else {
			Method method = new Method();
			method.setStandardName("Add" + this.domain.getCapFirstDomainName());
			method.addSignature(new Signature(1,"Json("+StringUtil.getSnakeName(this.domain.getStandardName())+"_request)","Json<"+this.domain.getCapFirstDomainName()+"Request>"));
			method.setReturnType(new Type("String"));
			
			List<Writeable> sList = new ArrayList<Writeable>();
			long serial = 1000L;
			for (Field f: this.domain.getFields()) {
				if (!"Image".equalsIgnoreCase(f.getFieldType())) {
					sList.add(new Statement(serial,1,"let "+f.getSnakeFieldName()+" = "+this.domain.getSnakeDomainName()+"_request."+f.getSnakeFieldName()+".unwrap_or_default();"));
					serial += 1000L;
				}
			}
			
			for (Field f: this.domain.getFields()) {
				if ("Image".equalsIgnoreCase(f.getFieldType())) {
					sList.add(new Statement(serial,1,"let my_"+f.getSnakeFieldName()+" = pick_"+this.domain.getSnakeDomainName()+"_picture(\""+f.getSnakeFieldName()+"\".to_string());"));
					serial += 1000L;
				}
			}
			
			sList.add(new Statement(serial,0,""));
			sList.add(new Statement(serial+1000L,1,"let "+this.domain.getSnakeDomainName()+" = "+this.domain.getCapFirstDomainNameWithSuffix()+"{"));
			serial += 2000L;
			for (Field f: this.domain.getFields()) {
				if (!"Image".equalsIgnoreCase(f.getFieldType())) {
					sList.add(new Statement(serial,1,""+f.getSnakeFieldName()+" : "+f.getSnakeFieldName()+","));
				}else {
					sList.add(new Statement(serial,1,""+f.getSnakeFieldName()+" : my_"+f.getSnakeFieldName()+","));
				}
				serial += 1000L;
			}
			sList.add(new Statement(serial,1,"};"));
			sList.add(new Statement(serial+1000L,1,""));
			
			sList.add(new Statement(serial+19000L,1,"let result = service_"+StringUtil.getSnakeName(this.getVerbName())+"("+this.domain.getSnakeDomainName()+").await;"));
			sList.add(new Statement(serial+20000L,1,"match result {"));
			sList.add(new Statement(serial+21000L,2,"Err(_) => {"));
			sList.add(new Statement(serial+22000L,3,"println!(\"Error!\");"));
			sList.add(new Statement(serial+23000L,3,"r#\"{  \"rows\": null,  \"success\": true}\"#.to_string()"));
			sList.add(new Statement(serial+24000L,2,"},"));
			sList.add(new Statement(serial+25000L,2,"Ok(result) => {"));
			sList.add(new Statement(serial+26000L,3,"let mut map = Map::new();"));
			sList.add(new Statement(serial+27000L,3,"map.insert(\"success\".to_string(), Value::from(true));"));
			sList.add(new Statement(serial+28000L,3,"map.insert("));
			sList.add(new Statement(serial+29000L,4,"\"rows\".to_string(),"));
			sList.add(new Statement(serial+30000L,4,"Value::from(\"\"),"));
			sList.add(new Statement(serial+31000L,3,");"));
			sList.add(new Statement(serial+32000L,3,""));
			sList.add(new Statement(serial+33000L,3,"let resultjson = serde_json::to_string_pretty(&map).unwrap();"));
			sList.add(new Statement(serial+34000L,3,"//println!(\"{}\", resultjson);"));
			sList.add(new Statement(serial+35000L,3,"return resultjson;"));
			sList.add(new Statement(serial+36000L,2,"}"));
			sList.add(new Statement(serial+37000L,1,"}"));

			method.setMethodStatementList(WriteableUtil.merge(sList));
			return method;
		}
	}

	@Override
	public JavascriptBlock generateEasyUIJSButtonBlock() throws Exception {
		if (this.denied)
			return null;
		else {
			JavascriptBlock block = new JavascriptBlock();
			block.setSerial(100);
			block.setStandardName("add" + domain.getCapFirstDomainName());
			StatementList sl = new StatementList();
			sl.add(new Statement(1000, 0, "{"));
			if (this.domain.getLanguage().equalsIgnoreCase("english")) {
				sl.add(new Statement(2000, 1, "text:'Add',"));
			} else {
				sl.add(new Statement(2000, 1, "text:'新增',"));
			}
			sl.add(new Statement(3000, 1, "iconCls:'icon-add',"));
			sl.add(new Statement(4000, 1, "handler:function(){"));
			sl.add(new Statement(5000, 2, "$('#wadd" + this.domain.getCapFirstDomainName() + "').window('open');"));
			sl.add(new Statement(6000, 1, "}"));
			sl.add(new Statement(7000, 0, "}"));
			block.setMethodStatementList(sl);
			return block;
		}
	}
	@Override
	public JavascriptMethod generateEasyUIJSActionMethod() throws Exception {
		if (this.denied)
			return null;
		else {
			Domain domain = this.domain;
			JavascriptMethod method = new JavascriptMethod();
			method.setSerial(200);
			method.setStandardName("add" + domain.getCapFirstDomainName());

			StatementList sl = new StatementList();
			sl.add(new Statement(1000, 1, "if ($(\"#ff\").form(\"validate\")) {"));
			sl.add(new Statement(2000, 2, "$.ajax({"));
			sl.add(new Statement(3000, 3, "type: \"post\","));
			sl.add(new Statement(4000, 3,
					"url: \"../" + this.domain.getControllerPackagePrefix() + domain.getLowerFirstDomainName()
							+ domain.getControllerNamingSuffix() + "/add" + domain.getCapFirstDomainName() + "\","));
			sl.add(new Statement(5000, 3, "data: JSON.stringify({"));
			long serial = 6000;
			for (Field f : domain.getFieldsWithoutId()) {
				if (f instanceof Dropdown) {
					sl.add(new Statement(serial, 3, f.getLowerFirstFieldName() + ":parseInt($(\"#ff\").find(\"#"
							+ f.getLowerFirstFieldName() + "\").combobox(\"getValue\")),"));
				} else if (f.getFieldType().equalsIgnoreCase("bool")) {
					sl.add(new Statement(serial, 3,
							f.getLowerFirstFieldName() + ":parseBoolean($(\"#ff\").find(\"input[name='"
									+ f.getLowerFirstFieldName() + "']:checked\").val()),"));
				} else if (f.getFieldType().equalsIgnoreCase("i32")||f.getFieldType().equalsIgnoreCase("i64")) {
					sl.add(new Statement(serial, 3, f.getLowerFirstFieldName() + ":parseInt($(\"#ff\").find(\"#"
							+ f.getLowerFirstFieldName() + "\").val()),"));
				} else if (f.getFieldType().equalsIgnoreCase("f64")||f.getFieldType().equalsIgnoreCase("f32")) {
					sl.add(new Statement(serial, 3, f.getLowerFirstFieldName() + ":parseFloat($(\"#ff\").find(\"#"
							+ f.getLowerFirstFieldName() + "\").val()),"));
				} else if (f.isTextarea()) {
					sl.add(new Statement(serial, 3, f.getLowerFirstFieldName() + ":$(\"#ff\").find(\"#"
							+ f.getLowerFirstFieldName() + "\").val(),"));
				} else {
					sl.add(new Statement(serial, 3, f.getLowerFirstFieldName() + ":$(\"#ff\").find(\"#"
							+ f.getLowerFirstFieldName() + "\").val(),"));
				}
				serial += 1000;
			}
			sl.add(new Statement(serial, 3, "}),"));
			sl.add(new Statement(serial + 1000, 3, "dataType: 'json',"));
			sl.add(new Statement(serial + 2000, 3, "contentType:\"application/json;charset=UTF-8\","));
			sl.add(new Statement(serial + 3000, 3, "success: function(data, textStatus) {"));
			sl.add(new Statement(serial + 4000, 4, "if (data.success) {"));
			sl.add(new Statement(serial + 5000, 5, "window.location.reload();"));
			sl.add(new Statement(serial + 9000, 4, "}"));
			sl.add(new Statement(serial + 10000, 4, "},"));
			sl.add(new Statement(serial + 11000, 3, "complete : function(XMLHttpRequest, textStatus) {"));
			sl.add(new Statement(serial + 12000, 3, "},"));
			sl.add(new Statement(serial + 13000, 3, "error : function(XMLHttpRequest,textStatus,errorThrown) {"));
			sl.add(new Statement(serial + 14000, 4, "alert(\"Error:\"+textStatus);"));
			sl.add(new Statement(serial + 15000, 4, "alert(errorThrown.toString());"));
			sl.add(new Statement(serial + 16000, 3, "}"));
			sl.add(new Statement(serial + 17000, 2, "});"));
			sl.add(new Statement(serial + 18000, 1, "}"));

			method.setMethodStatementList(sl);
			return method;
		}
	}

	public Set<Field> getDeniedFields() {
		return deniedFields;
	}

	public void setDeniedFields(Set<Field> deniedFields) {
		this.deniedFields = deniedFields;
	}

}
